package com.eii.resubmit.aop.request.annotation;

import org.springframework.core.annotation.AnnotationUtils;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.lang.reflect.Method;

/**
 * 请求限流注解
 * @author eii
 * @sice 2020/11/28
 */
@Target({ ElementType.METHOD, ElementType.TYPE })
@Retention(RetentionPolicy.RUNTIME)
public @interface LimitResquest {
    /**
     * 每秒每token限制请求数,默认值1.0
     */
    double value() default 1.0;
    /**
     * 超时时间,默认值3000毫秒
     */
    long timeout() default 3000;

    public static class LimitResquestHelper {
        public static LimitResquest get(Method method, Class<?> targetClass) {
            LimitResquest annotation = AnnotationUtils.findAnnotation(method, LimitResquest.class);
            if (annotation == null) {
                annotation = AnnotationUtils.findAnnotation(targetClass, LimitResquest.class);
            }
            return annotation;
        }
    }

}
